package main.java.com.amanda.suaFa.suafaAll;

import edu.princeton.cs.algs4.StdIn;
import edu.princeton.cs.algs4.StdOut;

import java.util.Iterator;
import java.util.NoSuchElementException;

/**
 * @author amanda
 * @Description 算法1.2 下压堆栈（链表实现
 *      这份泛型的Stack实现的基础是链表的数据结构，用于创建任意类型的栈
 */
public class Stack<Item> implements Iterable<Item> {
    private class Node {
        // 定义了结点的嵌套类
        Item item;
        Node next;
    }

    private Node first; // 栈顶
    private int N; // 元素数量

    public boolean isEmpty() {
        return first == null;
    }
    public int size() {
        return N;
    }
    public void push(Item item) {
        Node oldFirst = first;
        first = new Node();
        first.item = item;
        first.next = oldFirst;
        N++;
    }
    public Item pop() {
        Item item = first.item;
        first = first.next;
        N--;
        return item;
    }
    public Item peek() {
        if (isEmpty()) throw new NoSuchElementException("Stack underflow");
        return first.item;
    }

    public static void main(String[] args) {
        // 创建一个栈并根据StdIn中的指示压入或弹出字符串
        Stack<String> s = new Stack<String>();
        while (!StdIn.isEmpty()) {
            String item = StdIn.readString();
            if (!item.equals("-"))
                s.push(item);
            else if (!s.isEmpty())
                StdOut.print(s.pop() + " 元素弹出");
        }
        StdOut.println("(" + s.size() + " left on stack)");
    }

    @Override
    public Iterator<Item> iterator() {
        return null;
    }
}
